package com.itheima.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.mapper.EmpMapper;
import com.itheima.pojo.*;
import com.itheima.service.DeptService;
import com.itheima.service.EmpExprService;
import com.itheima.service.EmpService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

@Service
public class EmpServiceImpl extends ServiceImpl<EmpMapper, Emp> implements EmpService {

    @Autowired
    private DeptService deptService;

    @Autowired
    private EmpExprService empExprService;

    @Override
    public PageResult<Emp> pageResult(EmpQueryParam param) {
        // 1.设置分页参数
        Page<Emp> page = Page.of(param.getPage(), param.getPageSize());
        page.addOrder(OrderItem.desc("update_time"));

        // 2.设置执行条件并查询
        page = lambdaQuery().like(param.getName() != null && !param.getName().isEmpty(), Emp::getName, param.getName())
                            .eq(param.getGender() != null, Emp::getGender, param.getGender())
                            .between(param.getBegin() != null && param.getEnd() != null, Emp::getEntryDate,
                                    param.getBegin(), param.getEnd())
                            .page(page);

        // 3.封装部门名称到员工列表中
        // 3.1 使用set集合收集员工的部门id
        Set<Integer> deptIds = page.getRecords().stream().map(Emp::getDeptId).collect(Collectors.toSet());

        // 3.2 将部门名称封装到部门列表中
        if (!deptIds.isEmpty()) {
            List<Dept> deptList = deptService.listByIds(deptIds);
            Map<Integer, String> deptMap = deptList.stream().collect(Collectors.toMap(Dept::getId, Dept::getName));
            page.getRecords().forEach(emp -> {
                emp.setDeptName(deptMap.getOrDefault(emp.getDeptId(), null));

            });
        }

        // 4.封装结果并返回
        return new PageResult<>(page.getTotal(), page.getRecords());
    }

    @Override
    @Transactional
    public void saveEmp(Emp emp) {
        // 添加员工基本信息
        emp.setCreateTime(LocalDateTime.now());
        emp.setUpdateTime(LocalDateTime.now());
        save(emp);

        // 添加员工工作信息
        emp.getExprList().forEach(empExpr -> {
            empExpr.setEmpId(emp.getId());
        });
        empExprService.saveBatch(emp.getExprList());
    }

    @Override
    public void updateEmp(Emp emp) {
        // 1.修改员工基本信息
        emp.setUpdateTime(LocalDateTime.now());
        updateById(emp);

        // 2.删除员工工作信息
        empExprService.remove(new LambdaQueryWrapper<EmpExpr>().eq(EmpExpr::getEmpId, emp.getId()));

        // 3.保存员工工作信息
        emp.getExprList().forEach(empExpr -> {
            empExpr.setEmpId(emp.getId());
        });
        empExprService.saveBatch(emp.getExprList());
    }

    @Override
    public Emp getEmpById(Integer id) {
        //  1.查询员工基本信息
        Emp emp = getById(id);

        // 2.查询员工工作信息
        List<EmpExpr> empExprList = empExprService.lambdaQuery().eq(EmpExpr::getEmpId, id).list();

        emp.setExprList(empExprList);

        return emp;
    }

    @Override
    public void removeEmpByIds(List<Integer> ids) {
        // 1.删除员工基本信息
        removeByIds(ids);

        // 2.删除员工工作信息
        empExprService.remove(new LambdaQueryWrapper<EmpExpr>().in(EmpExpr::getEmpId, ids));
    }
}
